// 简单的查询测试脚本
import { searchLancedb } from './src/core/embed.js';

async function testQuery() {
  try {
    // 测试语义搜索
    const queryText = "创建数据库连接";
    const topK = 5;
    console.log(`\n语义搜索 "${queryText}" 的前 ${topK} 个结果:`);
    const vectorResults = await searchLancedb(queryText, topK);
    
    if (vectorResults && vectorResults.length > 0) {
      console.log(`找到 ${vectorResults.length} 个匹配结果:`);
      vectorResults.forEach((result, index) => {
        console.log(`\n结果 ${index + 1}:`);
        // 限制输出长度以便查看
        const displayText = result.length > 300 ? result.substring(0, 300) + '...' : result;
        console.log(displayText);
      });
    } else {
      console.log('没有找到匹配的结果');
    }
  } catch (error) {
    console.error("查询时出错:", error);
  }
}

testQuery();